1926B - Vlad and Shapes - CodeForces Solution

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int n;
        cin>>n;
        vector<int>v;
        for(int i=0;i<n;i++)
        {
            string s;
            cin>>s;
            int sum=0;
            for(int i=0;i<n;i++)
            {
                if(s[i]=='1')
                sum++;
            }
            if(sum!=0)
            v.push_back(sum);
            sum=0;
        }
        int u=0;
        for(int i=0;i<v.size()-1;i++)
        {
            if(v[i]!=v[i+1])
            {
                u=-1;
                break;
            }
        }
        if(u==-1)
        cout<<"TRIANGLE\n";
        else
        cout<<"SQUARE\n";
    }
}